generate 2d array in python